import chromadb

client=chromadb.Client()
collection=client.get_or_create_collection(name="demo")

document=["机器学习是一种实现人工智能的方法","深度学习是机器学习算法之一","大模型使用的就是深度学习方法"]
collection.add(documents=document,ids=["id1","id2","id3"])

# 相似度搜素
result= collection.query(query_texts=["什么是深度学习？"],n_results=2)
print(result["documents"][0][0])